#include <bits/stdc++.h>
using namespace std;

/*
2125. 银行中的激光束数量
已解答
中等
相关标签
premium lock icon
相关企业
提示
银行内部的防盗安全装置已经激活。给你一个下标从 0 开始的二进制字符串数组 bank ，表示银行的平面图，这是一个大小为 m x n 的二维矩阵。 bank[i] 表示第 i 行的设备分布，由若干 '0' 和若干 '1' 组成。'0' 表示单元格是空的，而 '1' 表示单元格有一个安全设备。

对任意两个安全设备而言，如果同时 满足下面两个条件，则二者之间存在 一个 激光束：

两个设备位于两个 不同行 ：r1 和 r2 ，其中 r1 < r2 。
满足 r1 < i < r2 的 所有 行 i ，都 没有安全设备 。
激光束是独立的，也就是说，一个激光束既不会干扰另一个激光束，也不会与另一个激光束合并成一束。

返回银行中激光束的总数量。

 

示例 1：



输入：bank = ["011001","000000","010100","001000"]
输出：8
解释：在下面每组设备对之间，存在一条激光束。总共是 8 条激光束：
 * bank[0][1] -- bank[2][1]
 * bank[0][1] -- bank[2][3]
 * bank[0][2] -- bank[2][1]
 * bank[0][2] -- bank[2][3]
 * bank[0][5] -- bank[2][1]
 * bank[0][5] -- bank[2][3]
 * bank[2][1] -- bank[3][2]
 * bank[2][3] -- bank[3][2]
注意，第 0 行和第 3 行上的设备之间不存在激光束。
这是因为第 2 行存在安全设备，这不满足第 2 个条件。
示例 2：



输入：bank = ["000","111","000"]
输出：0
解释：不存在两个位于不同行的设备
 

提示：

m == bank.length
n == bank[i].length
1 <= m, n <= 500
bank[i][j] 为 '0' 或 '1'
*/

// 法一
class Solution {
public:
    int numberOfBeams(vector<string>& bank) {
        vector<int> deviceCounts;   // 每行安全设备的数量

        // 每一行的安全设备数量 要过滤
        for (const string& row : bank) {
            // 计算当前行中 '1'
            int cnt = std::count(row.begin(), row.end(), '1');
            if (cnt > 0)  deviceCounts.push_back(cnt);
        }

        // affect not 2 --> 0
        if (deviceCounts.size() < 2)    return 0;

        // sum affect count
        int total = 0;
        for (int i = 1; i < deviceCounts.size(); i++) {
            // i - 1 与 i 之间 = 两行设备数乘积
            total += deviceCounts[i - 1] * deviceCounts[i];
        }
        return total;
    }
};

// 法二
class Solution {
public:
    int numberOfBeams(vector<string>& bank) {
        int last = 0;  // 上一行的设备数
        int ans = 0;   // 激光束总数
        
        for (const string& line : bank) {
            // 统计当前行的设备数（'1'的数量）
            int cnt = count_if(line.begin(), line.end(), [](char ch) { return ch == '1'; });
            
            if (cnt != 0) {  // 当前行有设备
                ans += last * cnt;  // 累加与上一行的激光束
                last = cnt;  // 更新上一行设备数为当前行
            }
        }
        return ans;
    }
};